#!/bin/bash
# ResNeXt 训练平台 - 生产模式启动脚本
# 此脚本确保应用以生产模式启动，禁用debug和代码重载

echo "==========================================="
echo "ResNeXt 训练平台 - 生产模式启动"
echo "==========================================="

# 设置生产环境变量
export FLASK_ENV=production
export PYTHONPATH="$(pwd)"

# 检查Python是否可用
if ! command -v python3 &> /dev/null && ! command -v python &> /dev/null; then
    echo "错误：Python未找到，请确保已安装Python"
    exit 1
fi

# 优先使用python3
PYTHON_CMD="python3"
if ! command -v python3 &> /dev/null; then
    PYTHON_CMD="python"
fi

# 检查Python版本
echo "检查Python版本..."
$PYTHON_CMD --version

# 检查必需的依赖
echo "检查依赖项..."
if ! $PYTHON_CMD -c "import torch, torchvision, flask, flask_socketio" &> /dev/null; then
    echo "错误：缺少必需的依赖项，请运行: pip install -r requirements.txt"
    exit 1
fi

echo "依赖检查完成"
echo ""

# 显示启动配置
echo "启动配置："
echo "- 模式：生产模式"
echo "- Debug：禁用"
echo "- 代码重载：禁用"
echo "- 端口：5001"
echo "- 主机：0.0.0.0"
echo ""

echo "正在启动ResNeXt训练平台..."
echo "访问地址：http://localhost:5001"
echo "按 Ctrl+C 停止服务器"
echo ""

# 启动应用
$PYTHON_CMD app.py

echo ""
echo "服务器已停止"
